Basic Cnn Model Using Pytorch With Image Augmentation